from dataclasses import dataclass
@dataclass
class ModelArgs:
    input_channels : int = 57 # 信道数量
    out_channels : int = 64
    output_dim : int = 30 # 分类数量
    input_dim : int = 400 
    input_dim2 : int = 256
    input_dim1 : int = 512
    device : str = 'cpu'
    alpha : float = 1.0
    warm_up : int = 2
    gpu_id : int = 4
    batch_size : int = 64
    epochs : int = 90
    learning_rate : float = 0.001
    nhead : int = 1
    num_encoder_layers : int = 2
    band_numbers : int = 4
    feature_size : int = 5
    save_dir : str = '/root/data/video_decoding/'